package com.longrise.android.loaddata.newloaddata;

import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.ArrayBlockingQueue;
import java.util.concurrent.BlockingQueue;
import java.util.concurrent.Future;
import java.util.concurrent.ThreadFactory;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicInteger;

/**
 * Created by chengsan on 16/5/20.
 */
public class ThreadPool
{
    /** 总共多少任务（根据CPU个数决定创建活动线程的个数,这样取的好处就是可以让手机承受得住） */
    private static final int CORE_POOL_SIZE = Runtime.getRuntime().availableProcessors() * 3 + 2;
    private static final String TAG = "ThreadPoolUtils";
    private static Map<Runnable, Future<?>> map = new HashMap<Runnable, Future<?>>();
    // 线程池核心线程数
    //private static int CORE_POOL_SIZE = 5;

    // 线程池最大线程数
    private static int MAX_POOL_SIZE = Runtime.getRuntime().availableProcessors() * 3 + 2;

    // 额外线程空状态生存时间
    private static int KEEP_ALIVE_TIME = 10000;

    // 阻塞队列。当核心线程都被占用，且阻塞队列已满的情况下，才会开启额外线程。
    private static BlockingQueue<Runnable> workQueue = new ArrayBlockingQueue<Runnable>(
            10);

    private ThreadPool()
    {

    }

    // 线程工厂
    private static ThreadFactory threadFactory = new ThreadFactory()
    {
        private final AtomicInteger integer = new AtomicInteger();

        @Override
        public Thread newThread(Runnable r)
        {
            return new Thread(r, "myThreadPool thread:" + integer.getAndIncrement());
        }
    };

    // 线程池
    private static ThreadPoolExecutor threadPool;

    static
    {
        threadPool = new ThreadPoolExecutor(CORE_POOL_SIZE, MAX_POOL_SIZE,
                KEEP_ALIVE_TIME, TimeUnit.SECONDS, workQueue, threadFactory);
    }

    /**
     * 从线程池中抽取线程，执行指定的Runnable对象
     * @param runnable
     */
    public static void execute(Runnable runnable)
    {
        Future<?> submit = threadPool.submit(runnable);
        map.put(runnable, submit);
    }

    public static void remove(Runnable runnable)
    {
        Future<?> submit = (Future<?>) map.get(runnable);
        if(submit!=null)
        {
            submit.cancel(true);
        }
    }
}
